Beat the Codility Coding Interview in Python